CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch --nproc_per_node=1 --master_port 44444 tools/dist_train.py \
--config_file='configs/softmax_triplet_pd.yaml' MODEL.DEVICE_ID "('0')" MODEL.TDeep "(0)" \
MODEL.PRETRAIN_PATH "('/home/ubuntu/yuyu/model_pretrain/resnet50-19c8e357.pth')" MODEL.NAME "('Pres50')" \
INPUT.SIZE_TRAIN "([256, 256])" INPUT.SIZE_TEST "([256, 256])" DATASETS.NAMES "('VeRi')"  \
OUTPUT_DIR "('/home/ubuntu/yuyu/gitee_UFDN/UFDN-Reid/cbaseline_veri_Pres50_decouple_loss_weight_0.25')" \
SOLVER.MAX_EPOCHS "(120)" MODEL.LOSS_LOCAL "(['focal',  'triplet'])" \
SOLVER.IMS_PER_BATCH "(64)" TEST.NECK_FEAT "('before')" TEST.FEAT_NORM "('yes')" DATALOADER.NUM_INSTANCE "16" \
SOLVER.STEPS "([40, 70])" SOLVER.WARMUP_ITERS "(10)" SOLVER.BASE_LR "(2e-4)" SOLVER.OPTIMIZER_NAME "('Adam')" SOLVER.scheduler_mode "('linear')"
# 81.5%  'triplet',